export dataset_name=/home/user4/workspace/proteingym-benchmark/data

# esm_if1_gvp4_t16_142M_UR50
# esm2_t6_8M_UR50D esm2_t12_35M_UR50D esm2_t30_150M_UR50D esm2_t33_650M_UR50D esm2_t36_3B_UR50D esm2_t48_15B_UR50D
# esm1v_t33_650M_UR90S_1 esm1v_t33_650M_UR90S_2 esm1v_t33_650M_UR90S_3 esm1v_t33_650M_UR90S_4 esm1v_t33_650M_UR90S_5
# esm1b_t33_650M_UR50S
CUDA_VISIBLE_DEVICES=6 python src/ESM/predict.py \
    --dataset_input $dataset_name \
    --model_location esm2_t33_650M_UR50D \
    --mutation_col mutant \
    --scoring_strategy masked-marginals \
    --use_sum